home *** CD-ROM | disk | FTP | other *** search
/ The Atari Compendium / The Atari Compendium (Toad Computers) (1994).iso / files / umich / network / ka9q / nhclb120.zoo / ftpserv.c < prev    next >
C/C++ Source or Header  |  1992-06-21  |  19KB  |  778 lines
  1. /* FTP Server state machine - see RFC 959 */
  2.  
  3. #define    LINELEN        128    /* Length of command buffer */
  4.  
  5. #include <stdio.h>
  6. #include <ctype.h>
  7. #include "global.h"
  8. #include "config.h"
  9. #include "mbuf.h"
  10. #include "netuser.h"
  11. #include "timer.h"
  12. #include "tcp.h"
  13. #include "ftp.h"
  14. #include "telnet.h"
  15. #include "iface.h"
  16. #include "ax25.h"
  17. #include "lapb.h"
  18. #include "finger.h"
  19. #include "session.h"
  20. #include "nr4.h"
  21. #ifdef UNIX
  22. #include "unix.h"
  23. #undef    toupper
  24. #undef    tolower
  25. #include <string.h>
  26. #include <sys/types.h>
  27. time_t  time();
  28. #endif /* UNIX */
  29. static void ftpscs();
  30. static void ftpscr();
  31. static void ftpcommand();
  32.  
  33. static int pport();
  34. static int ftplogin();
  35.  
  36. /* Command table */
  37. static char *commands[] = {
  38.     "user",
  39. #define    USER_CMD    0
  40.     "acct",
  41. #define    ACCT_CMD    1
  42.     "pass",
  43. #define    PASS_CMD    2
  44.     "type",
  45. #define    TYPE_CMD    3
  46.     "list",
  47. #define    LIST_CMD    4
  48.     "cwd",
  49. #define    CWD_CMD        5
  50.     "dele",
  51. #define    DELE_CMD    6
  52.     "name",
  53. #define    NAME_CMD    7
  54.     "quit",
  55. #define    QUIT_CMD    8
  56.     "retr",
  57. #define    RETR_CMD    9
  58.     "stor",
  59. #define    STOR_CMD    10
  60.     "port",
  61. #define    PORT_CMD    11
  62.     "nlst",
  63. #define    NLST_CMD    12
  64.     "pwd",
  65. #define    PWD_CMD        13
  66.     "xpwd",            /* For compatibility with 4.2BSD */
  67. #define    XPWD_CMD    14
  68.     "mkd ",
  69. #define    MKD_CMD        15
  70.     "xmkd",            /* For compatibility with 4.2BSD */
  71. #define    XMKD_CMD    16
  72.     "xrmd",            /* For compatibility with 4.2BSD */
  73. #define    XRMD_CMD    17
  74.     "rmd ",
  75. #define    RMD_CMD        18
  76.     "stru",
  77. #define    STRU_CMD    19
  78.     "mode",
  79. #define    MODE_CMD    20
  80.     "help",
  81. #define HELP_CMD        21
  82.     NULLCHAR
  83. };
  84.  
  85. /* Response messages */
  86. static char helpmsg[]=  "214-Commands:\r\n\
  87. 214-\r\n\
  88. 214-USER ACCT PASS TYPE LIST CWD  DELE NAME QUIT RETR STOR PORT\r\n\
  89. 214-NLST PWD  XPWD MKD  XMKD XRMD RMD  STRU MODE HELP\r\n\
  90. 214 End\r\n";
  91. static char banner[] = "220 %s FTP version %s ready at %s\r\n";
  92. static char badcmd[] = "500 Unknown command\r\n";
  93. static char unsupp[] = "500 Unsupported command or option\r\n";
  94. static char givepass[] = "331 Enter PASS command\r\n";
  95. static char logged[] = "230 Logged in\r\n";
  96. static char typeok[] = "200 Type OK\r\n";
  97. static char only8[] = "501 Only logical bytesize 8 supported\r\n";
  98. static char deleok[] = "250 File deleted\r\n";
  99. static char mkdok[] = "200 MKD ok\r\n";
  100. static char delefail[] = "550 Delete failed\r\n";
  101. static char pwdmsg[] = "257 \"%s\" is current directory\r\n";
  102. static char badtype[] = "501 Unknown type \"%s\"\r\n";
  103. static char badport[] = "501 Bad port syntax\r\n";
  104. static char unimp[] = "502 Command not yet implemented\r\n";
  105. static char bye[] = "221 Goodbye!\r\n";
  106. static char nodir[] = "553 Can't read directory \"%s\"\r\n";
  107. static char cantopen[] = "550 Can't read file \"%s\"\r\n";
  108. static char sending[] = "150 Opening data connection for %s %s\r\n";
  109. static char cantmake[] = "553 Can't create \"%s\"\r\n";
  110. static char portok[] = "200 Port command okay\r\n";
  111. static char rxok[] = "226 File received OK\r\n";
  112. static char txok[] = "226 File sent OK\r\n";
  113. static char noperm[] = "550 Permission denied\r\n";
  114. static char noconn[] = "425 Data connection reset\r\n";
  115. static char notlog[] = "530 Please log in with USER and PASS\r\n";
  116. static char okay[] = "200 Ok\r\n";
  117.  
  118. static struct tcb *ftp_tcb;
  119.  
  120. /* Start up FTP service */
  121. ftp1(argc,argv)
  122. int argc;
  123. char *argv[];
  124. {
  125.     struct socket lsocket;
  126.     void ftpscr(),ftpscs();
  127.  
  128.     lsocket.address = ip_addr;
  129.     if(argc < 2)
  130.         lsocket.port = FTP_PORT;
  131.     else
  132.         lsocket.port = atoi(argv[1]);
  133.  
  134.     ftp_tcb = open_tcp(&lsocket,NULLSOCK,TCP_SERVER,0,ftpscr,NULLVFP,ftpscs,0,(char *)NULL);
  135. }
  136. /* Shut down FTP server */
  137. ftp0()
  138. {
  139.     if(ftp_tcb != NULLTCB)
  140.         close_tcp(ftp_tcb);
  141. }
  142. /* FTP Server Control channel State change upcall handler */
  143. /*ARGSUSED*/
  144. static
  145. void
  146. ftpscs(tcb,old,new)
  147. struct tcb *tcb;
  148. char old,new;
  149. {
  150.     extern char hostname[],version[];
  151.     struct ftp *ftp,*ftp_create();
  152.     void ftp_delete();
  153.     char *inet_ntoa();
  154.     long t;
  155.     char *cp,*cp1;
  156.  
  157.     switch(new){
  158. /* Setting QUICKSTART piggybacks the server's banner on the SYN/ACK segment;
  159.  * leaving it unset waits for the three-way handshake to complete before
  160.  * sending the banner. Piggybacking unfortunately breaks some old TCPs,
  161.  * so its use is not (yet) recommended.
  162. */
  163. #ifdef    QUICKSTART
  164.     case SYN_RECEIVED:
  165. #else
  166.     case ESTABLISHED:
  167. #endif
  168.         if((ftp = ftp_create(LINELEN)) == NULLFTP){
  169.             /* No space, kill connection */
  170.             close_tcp(tcb);
  171.             return;
  172.         }
  173.         ftp->control = tcb;        /* Downward link */
  174.         tcb->user = (char *)ftp;    /* Upward link */
  175.  
  176.         /* Set default data port */
  177.         ftp->port.address = tcb->conn.remote.address;
  178.         ftp->port.port = FTPD_PORT;
  179.  
  180.         /* Note current directory */
  181.         log(tcb,"open FTP");
  182.         time(&t);
  183.         cp = ctime(&t);
  184.         if((cp1 = index(cp,'\n')) != NULLCHAR)
  185.             *cp1 = '\0';
  186.         tprintf(ftp->control,banner,hostname,version,cp);
  187.         break;        
  188.     case CLOSE_WAIT:
  189.         close_tcp(tcb);
  190.         break;
  191.     case CLOSED:
  192.         log(tcb,"close FTP");
  193.         if((ftp = (struct ftp *)tcb->user) != NULLFTP ||
  194.            !tcpval(ftp->control)) {   /* control session existing ? */
  195.             if (ftp != NULLFTP)
  196.                 ftp_delete(ftp);
  197.         }
  198.         /* Check if server is being shut down */
  199.         if(tcb == ftp_tcb)
  200.             ftp_tcb = NULLTCB;
  201.         del_tcp(tcb);
  202.         break;
  203.     }
  204. }
  205.  
  206. /* FTP Server Control channel Receiver upcall handler */
  207. /*ARGSUSED*/
  208. static
  209. void
  210. ftpscr(tcb,cnt)
  211. struct tcb *tcb;
  212. int16 cnt;
  213. {
  214.     register struct ftp *ftp;
  215.     char c;
  216.     struct mbuf *bp;
  217.     void ftpcommand();
  218.  
  219.     if((ftp = (struct ftp *)tcb->user) == NULLFTP){
  220.         /* Unknown connection, just kill it */
  221.         close_tcp(tcb);
  222.         return;
  223.     }
  224.     switch(ftp->state){
  225.     case COMMAND_STATE:
  226.         /* Assemble an input line in the session buffer. Return if incomplete */
  227.         recv_tcp(tcb,&bp,0);
  228.         while(pullup(&bp,&c,1) == 1){
  229.             switch(c){
  230.             case '\r':    /* Strip cr's */
  231.                 continue;
  232.             case '\n':    /* Complete line; process it */
  233.                 ftp->buf[ftp->cnt] = '\0';
  234.                 ftpcommand(ftp);
  235.                 ftp->cnt = 0;
  236.                 break;
  237.             default:    /* Assemble line */
  238.                 if(ftp->cnt != LINELEN-1)
  239.                     ftp->buf[ftp->cnt++] = c;
  240.                 break;
  241.             }
  242.         }
  243.         /* else no linefeed present yet to terminate command */
  244.         break;
  245.     case SENDING_STATE:
  246.     case RECEIVING_STATE:
  247.         /* Leave commands pending on receive queue until
  248.          * present command is done
  249.          */
  250.         break;
  251.     }
  252. }
  253.  
  254. /* FTP server data channel connection state change upcall handler */
  255. void
  256. ftpsds(tcb,old,new)
  257. struct tcb *tcb;
  258. char old,new;
  259. {
  260.     register struct ftp *ftp;
  261.  
  262.     if((ftp = (struct ftp *)tcb->user) == NULLFTP){
  263.         /* Unknown connection. Kill it */
  264.         del_tcp(tcb);
  265.     } else if((old == FINWAIT1 || old == CLOSING) && ftp->state == SENDING_STATE){
  266.         /* We've received an ack of our FIN while sending; we're done */
  267.         ftp->state = COMMAND_STATE;
  268.         tprintf(ftp->control,txok);
  269.         /* Kick command parser if something is waiting */
  270.         if(ftp->control->rcvcnt != 0)
  271.             ftpscr(ftp->control,ftp->control->rcvcnt);
  272.     } else if(ftp->state == RECEIVING_STATE && new == CLOSE_WAIT){
  273.         /* FIN received on incoming file */
  274. #ifdef    CPM
  275.         if(ftp->type == ASCII_TYPE)
  276.             putc(CTLZ,ftp->fp);
  277. #endif
  278.         close_tcp(tcb);
  279.         if(ftp->fp != stdout)
  280.             fclose(ftp->fp);
  281.         ftp->fp = NULLFILE;
  282.         ftp->state = COMMAND_STATE;
  283.         tprintf(ftp->control,rxok);
  284.         /* Kick command parser if something is waiting */
  285.         if(ftp->control->rcvcnt != 0)
  286.             ftpscr(ftp->control,ftp->control->rcvcnt);
  287.     } else if(new == CLOSED){
  288.         if(tcb->reason != NORMAL){
  289.             /* Data connection was reset, complain about it */
  290.             tprintf(ftp->control,noconn);
  291.             /* And clean up */
  292.             if(ftp->fp != NULLFILE && ftp->fp != stdout)
  293.                 fclose(ftp->fp);
  294.             ftp->fp = NULLFILE;
  295.             ftp->state = COMMAND_STATE;
  296.             /* Kick command parser if something is waiting */
  297.             if(ftp->control->rcvcnt != 0)
  298.                 ftpscr(ftp->control,ftp->control->rcvcnt);
  299.         }
  300.         /* Clear only if another transfer hasn't already started */
  301.         if(ftp->data == tcb)
  302.             ftp->data = NULLTCB;
  303.         del_tcp(tcb);
  304.     }
  305. }
  306.  
  307. /* Parse and execute ftp commands */
  308. static
  309. void
  310. ftpcommand(ftp)
  311. register struct ftp *ftp;
  312. {
  313.     void ftpdr(),ftpdt(),ftpsds();
  314.     char *cmd,*arg,*cp,**cmdp,*file;
  315.     char *pathname();
  316.     char *mode;
  317.     struct socket dport;
  318.     int i;
  319.  
  320. #ifndef    CPM
  321.     FILE *dir();
  322. #endif
  323.  
  324.     cmd = ftp->buf;
  325.     if(ftp->cnt == 0){
  326.         /* Can't be a legal FTP command */
  327.         tprintf(ftp->control,badcmd);
  328.         return;
  329.     }    
  330.     cmd = ftp->buf;
  331.  
  332. #ifdef    UNIX
  333.     /* Translate first word to lower case */
  334.     for(cp = cmd;*cp != ' ' && *cp != '\0';cp++)
  335.         *cp = tolower(*cp);
  336. #else
  337.     /* Translate entire buffer to lower case */
  338.     for(cp = c